from llama_index.core import VectorStoreIndex, SimpleDirectoryReader

# 从data目录加载所有的文档 支持格式的文档 docx ppt excel pdf都支持
documents = SimpleDirectoryReader("data").load_data()
# 使用加载的文档创建存储索引 在内部会自动将文档切成块并生成向量索引
index = VectorStoreIndex.from_documents(documents)
# 从索引创建查询引擎
query_engine = index.as_query_engine()
# 向查询引擎提问获取回答
response = query_engine.query("红楼梦的主要人物介绍?")
print(response)
